import tensorflow as tf

# 加载原始的 Keras 模型
model = tf.keras.models.load_model("model/keras_model.h5")

# 使用 TFLiteConverter 将模型转换为 TensorFlow Lite 格式
converter = tf.lite.TFLiteConverter.from_keras_model(model)
tflite_model = converter.convert()

# 将转换后的 TFLite 模型保存为 .tflite 文件
with open("model/keras_model.tflite", "wb") as f:
    f.write(tflite_model)